// Copyright (c) 2025 Bytedance Ltd. and/or its affiliates
// 
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
// 
//     http://www.apache.org/licenses/LICENSE-2.0
// 
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

import { EvalPlugin, EvaluatorConfig } from '@web-bench/evaluator-types'
import { ReportPlugin } from './reporter'
import { BenchProjectPlugin } from './bench'
import { BasePlugin } from './base'
import { CustomProjectPlugin } from './custom'

const InternalPlugins = [BasePlugin, BenchProjectPlugin, ReportPlugin, CustomProjectPlugin]

export const getDefaultPlugins = (projectType: EvaluatorConfig['projectType']): EvalPlugin[][] => {
  if (projectType === 'bench') {
    return [BasePlugin, BenchProjectPlugin, ReportPlugin]
  }
  return [BasePlugin, CustomProjectPlugin]
}

export const getInternalPluginsByName = (name: string): EvalPlugin[] => {
  return InternalPlugins.flat().filter((v) => v.name === name)
}
